package com.c2b.algorithm.leetcode.base;

import java.util.HashMap;
import java.util.Map;

/**
 * <a href='https://leetcode.cn/problems/check-array-formation-through-concatenation/'>能否连接形成数组(Check Array Formation Through Concatenation)</a>
 * <p>给你一个整数数组 arr ，数组中的每个整数 互不相同 。另有一个由整数数组构成的数组 pieces，其中的整数也 互不相同 。请你以 任意顺序 连接 pieces 中的数组以形成 arr 。但是，不允许 对每个数组 pieces[i] 中的整数重新排序。</p>
 * <p>如果可以连接 pieces 中的数组形成 arr ，返回 true ；否则，返回 false 。</p>
 *
 * <p>
 * <b>示例：</b>
 * <pre>
 * 示例 1：
 *      输入：arr = [15,88], pieces = [[88],[15]]
 *      输出：true
 *      解释：依次连接 [15] 和 [88]
 *
 * 示例 2：
 *      输入：arr = [49,18,16], pieces = [[16,18,49]]
 *      输出：false
 *      解释：即便数字相符，也不能重新排列 pieces[0]
 *
 * 示例 3：
 *      输入：arr = [91,4,64,78], pieces = [[78],[4,64],[91]]
 *      输出：true
 *      解释：依次连接 [91]、[4,64] 和 [78]
 * </pre>
 * </p>
 *
 * <p>
 * <b>提示：</b>
 *     <ul>
 *         <li>1 <= pieces.length <= arr.length <= 100</li>
 *         <li>sum(pieces[i].length) == arr.length</li>
 *         <li>1 <= pieces[i].length <= arr.length</li>
 *         <li>1 <= arr[i], pieces[i][j] <= 100</li>
 *         <li>arr 中的整数 互不相同</li>
 *         <li>pieces 中的整数 互不相同（也就是说，如果将 pieces 扁平化成一维数组，数组中的所有整数互不相同）</li>
 *     </ul>
 * </p>
 *
 * @author c2b
 * @since 2024/1/15 10:50
 */
public class LC1640CheckArrayFormationThroughConcatenation_S {
    static class Solution {
        public boolean canFormArray(int[] arr, int[][] pieces) {
            Map<Integer, int[]> map = new HashMap<>(pieces.length);
            for (int[] piece : pieces) {
                map.put(piece[0], piece);
            }
            int currIndex = 0;
            while (currIndex < arr.length) {
                if (!map.containsKey(arr[currIndex])) {
                    return false;
                }
                int[] ints = map.get(arr[currIndex]);
                for (int anInt : ints) {
                    if (arr[currIndex] != anInt) {
                        return false;
                    }
                    ++currIndex;
                }
            }
            return true;
        }
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        System.out.println(solution.canFormArray(new int[]{15, 88}, new int[][]{{88}, {15}}));
        System.out.println(solution.canFormArray(new int[]{49, 18, 16}, new int[][]{{16, 18, 49}}));
        System.out.println(solution.canFormArray(new int[]{91, 4, 64, 78}, new int[][]{{78}, {4, 64}, {91}}));
    }
}
